from api.ihrmEmpApi import IhrmEmpApi
from common.assert_tools import assert_use
from common.db_tools import DBTools
from common.get_header_tools import get_header
from config import TEL


# 定义测试类
class TestAddEmp(object):

    def setup_class(self):
        self.headers = get_header()

    def setup(self):
        # 调 数据库工具方法， 删除 指定手机号
        DBTools.uid_db(f"delete from bs_user where mobile = '{TEL}';")

    def teardown(self):
        # 调 数据库工具方法， 删除 指定手机号
        DBTools.uid_db(f"delete from bs_user where mobile = '{TEL}';")

    # 定义测试方法 - 添加成功(必选)
    def test01_add_ok(self):
        # 调用 自己封装的 添加员工接口

        req_data = {
            "username": "zs4",
            "mobile": TEL,
            "workNumber": "9527"
        }
        resp = IhrmEmpApi.add_emp(self.headers, req_data)
        print("添加员工成功：", resp.json())
        # 断言
        assert_use(resp, 200, True, 10000, "操作成功")

    # 定义测试方法 - 少参
    def test02_less_params(self):
        # 调用 自己封装的 添加员工接口
        req_data = {
            "mobile": TEL,
            "workNumber": "9527"
        }
        resp = IhrmEmpApi.add_emp(self.headers, req_data)
        print("少参：", resp.json())
        # 断言
        assert_use(resp, 200, False, 20002, "新增员工失败")
